Add some stuff

Fixe some "bug"