Add size classes in template instead
Tests to check adding size classes