Code correction...

...bet it still doesn't work though