Added base code for problem 3