If/elif/else syntax as well as building a function in python