got the model working
adding more UI