trying to fix week4