Minor corrections

minor stuff less bugs :)