minor changes
It works!