fixed solution
fixed homework