code cleaned up and example picture added