Added prediction feature functionality.
Merge pull request #9 from rhernandez04/master

Added prediction feature functionality.