Added formatted data and simple prediction script