Adding a predict method
Added clarifying comment to how predict is not yet optimal.