Added updates to test