nits + fix fig 5