added square function - chapter 03 - second edition