Wrote function to get weights, started main file