Cleanup and add comments
Added readme