Rewrite setup using 'functions'