Change some code in set_up_site