Fixed code

Added print statements