Added a sub module for handling stage related functions