My implementation of the Square challenge