Removed print statements and fixed some stuff.