Added figure capture
Updated readme