Adding hi message, and fixing unit test for square