Making toggle work on ajax cases