Solution changes for previous commit

oops