Fixing setup & docs