Removed unused import
Added method for getting entity amount. Closes #34