Added predict method (doesn't appear to work well yet though)