Cleaning up code
Formatted plot