Add polygon class
Add line class