Added code so users can (almost) choose their task mode

Notice that this isn't fully implemented yet.