few small changes
fixed it