Got If statements working correctly