small fixes
added some captions, fixed some small bugs