got square working with tests