Finalized the README + some comments