Worked on dialogue box functionality

Added models, worked on dialogue box.