Separate out driver code