Add install code to right file