better code

using a generator instead of this long integer list.