Add more menu functionality