Add missing fields to model
Fixed Session Tests