Added testing for staged files