minor update
added testing