Modify to work(this require_relative is too bad)