finishing touches

created if statements