A Punnett square can help demonstrate independent assortment, but the best way to demonstrate is to first identify the four possible combinations of alleles that can occur in gametes during meiosis.
If these alleles assort independently, there is an equal chance for each of these combinations:
A1B1, A1B2, A2B1, A2B2 (25% chance for each of these combinations)