HackerRank Day 1: Functions problem solution - TheCScience