Customised the datamodel by adding code to the db context