Added draw rectangle method and Diamond class