Got ViewModels going working on DB adding