Putting the code in the right place