Updated prob61.py comments