Lab 7 Code

Cleaned code for next step