Rename main functions for each stage