Removed print statements and added a missing field