Add inference example of squares