Put code in the right place