Added figures and docs