add square class