Update readme and closes #2