part 1 done (made simple changes)