Made the code example look a bit nicer