Added prediction prototype