add else/if statement to see if it works