how to write better python