made it a little cleaner to read
cleaned code