NCPC2021 Warmup #8 Code@LTH