Add environment module for migrations