updated readme and fixed some imports