Prettier Format Python Function Return