added more documentation and figures