clean up some code, added images