Having some little problems with if stmts