First working version with two grain layers