Minor update

minor update + test