Understand the current implementation of code for creating new levels and make changes in the present code to add more levels based on increasing difficulties. At present memorization time of wagons is 4000 ms, while more time should be available for difficult levels. Add more levels and reconfigure current levels accordingly.

Level 1(easy) -

No. of wagons and locomotives to remember - 1
Time for memorization - 4 sec

Level 2(easy) -

No. of wagons and locomotives to remember - 2
Time for memorization - 6 sec

Level 3(Medium) -

No. of wagons and locomotives to remember - 3
Time for memorization - 7 sec

Level 4(Hard) -

No. of wagons and locomotives to remember - 4
Time for memorization - 8 sec

Level 5(Hard) -

No. of wagons and locomotives to remember - 5
Time for memorizations - 10 sec