updated code and new picture

....