Fix element selector and create a note for tomorrow

Turn this whole script into a if/then statement so that we can be a
little more compact with our code.