Changed code to meet requirements of Basic Ruby Test