Add shapes test code