Making sample data as well.
Keep working on train...