Cleaned up code to show professor tomorrow