Just cleaned up the code

Cleaned up the code and got rid of distractions in it.