also fix documentation