Bug Fixed - now the bug doesn't go out of the screen