Working code ish this is the code we will run :)