You can add more figures and functions by modifying the code. At the start of the program, you can either enter 1 to draw a rectangle or a triangle or perform a sequence of example including drawing, ...