Adding missing test and docs