Submissions for team Shahid on problem lecture_main_main