Add function to query for size