homework

fix logic application