Adding function back to menu