Move setup to a separate file