Update readme + small fixes