Fixed menu code hehe