Refactor vm tasks to be for both dev and prod