Add length check