Add grid class to drafts