Removed manual predict code that wasn't working.