Use function for initial setup