Added SetMenu function