changed model names and added dropdown menu