Fixed imports and readme