Issue #73 Add model tests

Issue #73 
@mnzaki please review