Adding menu code