Added some bitchin' prediction code