Initial commit, solution for problem 1