Added stuff to do a simple test