Cleaned things up a bit more by removing unneeded logic, now that we have a function here, and also added a try except statement.