finished initial UI - now need to work on model