Return done instead of undefined across JS