$(function() {
    $.each($('ul#nav').find('ul'), function(i, el){
      if($(el).find('a.current').length > 0)
      {
        $(el).find('a.current').parents('ul').css({'display': 'block'});
        $(el).find('a.current').parents('li').addClass('selected');
      }
      else
      {
        $(el).css({'display': 'none'});
      }
    });
    

    $.each($('ul#nav').find('li'), function(i, el){
      if($(el).find('ul:first').length > 0)
        {
          var parent_link = $(el).find('a').get(0);
          $(parent_link).click(function(){            
            var nav_list = $(this).parent().find('ul');

            // reset selected state
            $("#nav li").removeClass("selected");
            $(this).parents('li').addClass("selected");

            if(nav_list.css('display') == 'block')
            {
              $(this).parent().find('ul:first').css({'display': 'none'});
            }
            else
            {
              //close all other lists
              $('ul#nav ul').css({'display': 'none'});
              $(this).parents('ul').css({'display': 'block'});
              $(this).parent().find('ul:first').css({'display': 'block'});
            }
            return false;
          });
        }
    });
});