Fixed documentation, updated test, still need to write tests