Adds wrapper for yolos