Fix assignment section sample code layout