Fixed trait not implementing the interface

#yolo

Oh and I changed ew code <3