Updated: made menu a directive