Update layout

I believe so. Pretty sure this line is added in chapter 5 of the book. Basically everything I've done so far is from the book, pretty much verbatim.