Added code for section02 problems