Question on hackerblocks

#1

question: https://hack.codingblocks.com/app/practice/2/641/problem
sol: https://ide.codingblocks.com/s/269538

changes in the code;