Add unit screenshot, updates to readme