Edit javascript to removeClass