Add install code