making the animations look a bit better, adding remove class