added some code to make it more interesting