Better code, working with more functions and object square.