Better types
Adding README
Minor fix