Cleaned up the code

Step 9 complete