Imported code from busbud-challenge repo