Cleaned up and commented code - professional