make it more pythony