Code and readme updated