add more solutions
idiomatic Python