Add (broken) code to draw the layers