Adding Model functionality