layer added (just draft!)

Train is a bit buggy!