write prettier python