Python easy coding