adding in my code from before