clean up imports and update readme