Stage 4 of Assignment