Edited printing statements in Python