Add setup function for application