Added functionality for stage loading