fixed my own mistakes
fixed import error