Smart class - Help me to suit my needs