Added main to problem 16