add docs + some minor changes