A short program for Hacker Rank