Add more luigi wrappers