Made print statements more intuitive and cleaned up comments