small changes
hope it works