Finish implementing code challenge