Readme updated. small code cleanup