Tweaked menu to work on li click