Adding import numpy as np after restructuring