Toggle instead of remove class