Use more list comprehensions
Use zip & *args