code cleanup and readme