Made the sample code look a little more simple