changed UI and added new model