Updated readme + comments