bugfix and docstrs