Improve wallJump