code improvements, readme updates