Update + fix docs