Final Sample Commit

Fixed the code with working models.