Lab 6 - Completed Lab Task - Clean up of code.

Removed unnecessary code.