Moved DbContext back to models