Gonna start a wrapper for all models out there