updated with fix to stage 3