If statement working a bit.