More readme + bug fix