New Feature (Additonal): Search Functionality

Fixed issue with setting stage size.