Add square class 

#117