Improve readme and setup