Use toggleClass instead of if-else
Better variable name