Updated readme and removed some imports