Starts to add some tests - woo