Cleaned up code

Deleted print statements
Remove #include iostream