Cleaned up the code after refactoring

Removed console logs and fixed some white space.