Menu added to existing code