Add drawing of classes. Draw rectangle, where user will click when button is selected