Removed some comments and fixed print statement