Cleaned prints and unused vars
Added some comments