Fix readme

Update the README a bit.