small changes
fixed: code