Updated readme
Cleaned up imports