$(document).ready(function() {
	$('li#category1 a').hover(function() {
 			$('li#category1').addClass('color');
	}, function() {
			$('li#category1').removeClass('color');
	});

	$('li#category2 a').hover(function() {
 			$('li#category2').addClass('color');
	}, function() {
			$('li#category2').removeClass('color');
	});

	$('li#category3 a').hover(function() {
 			$('li#category3').addClass('color');
	}, function() {
			$('li#category3').removeClass('color');
	});

	$('li#category4 a').hover(function() {
 			$('li#category4').addClass('color');
	}, function() {
			$('li#category4').removeClass('color');
	});

	$('li#category5 a').hover(function() {
 			$('li#category5').addClass('color');
	}, function() {
			$('li#category5').removeClass('color');
	});

	$('li#category6 a').hover(function() {
 			$('li#category6').addClass('color');
	}, function() {
			$('li#category6').removeClass('color');
	});
});

jQuery.fn.initMenu = function(){
    return this.each(function(){
        var theMenu = $(this).get(0);
        $('.sub-menu', this).hide();
		$('li.current-menu-ancestor').children('ul.sub-menu').show();
		$('li.current-menu-ancestor').children('ul.sub-menu').addClass('active');
		$('li.current-page-ancestor').parents('ul.sub-menu').show();
		$('li.current-page-ancestor').parents('ul.sub-menu').addClass('active');
        $('li.expand > .sub-menu', this).show();
        $('li.expand > .sub-menu', this).prev().addClass('active');
        $('li a', this).click(function(e){
            e.stopImmediatePropagation();
            var theElement = $(this).next();
            var parent = $(this).closest('ul');
            if (theElement.hasClass('sub-menu') && theElement.is(':visible')) {
                console.log(parent.html());
                 if (parent.hasClass('collapsible')) {
                        $('.sub-menu:visible', parent).first().slideUp('normal', function(){
                            $(this).prev().removeClass('active');
                        });
                        return false;
                    }
                    return false;
            }
            if (theElement.hasClass('sub-menu') && !theElement.is(':visible')) {

                //custom - adds class at beginning of expansion
                $(this).addClass('active');

                $('.sub-menu:visible', parent).first().slideUp('normal', function(){
                    $(this).prev().removeClass('active');
                });
                theElement.slideDown('normal', function(){
                      $(this).prev().addClass('active');
                 });
                return false;
             }
        });
    });
};
$(document).ready(function(){
    $('.menu').initMenu();
});