Add class Square and restructure