La till Entity klass (långt ifrån klar) :D