Updated docs and fixed last import