Assignment - Functions

if and else if updated, still trying to figure out how to work in a
function