minor bug fix and added images