adding variables and experimenting with the return
getting errors, fixing