Made some changes (print statements, fixed an error, etc.)