Fixed issue #3 by adding if statement to catch if a variable passed in or not.