Fixed bug; updated docs