removed Line 'class' and replaced it with Rectangle