Ammended if-else statement, need to add 3 more modules and work on else
if