Adding square class