Update readme + small changes