Update test file and add docs