added square class