Update square.py

Just added docstrings to square and cube functions