Will also look at a function to make this re-usable