Make it pythonic