Added some neat test code :)