V0.1

Working model with batch size 1000