Learning by doing. Updated test to work properly.