Update README + small fixes