More compact python