try to fix Bonus problem set