Reformat get_elevator code. Used built in python functions to shorten some ugly code.