Played around with some of the code