updated the final test - it all works