updates to issue 5, fixes image issue