added imports to main
updated readme