Adding assignment feature to menu