Stage 2 of assignment complete

All the functionality for stage 2 submission completed