Signed-off-by: s3316048 <s3316048@student.rmit.edu.au>
add some codes in main class