2nd try on fixing problems for part b