issue 2 finished

please help me review the code