Adding if statement

Will not error if input is none