Organized imports and updated the README