Fixed some import errors.
Added main python file