Fix some stuff and add
Update readme