Making sure to use .loc everywhere