I have added in  models two classes, and a added a function in views

@iBebarz @nadeembarakat @saraismail1234 @mohamedhammad89 please review my code