Code format, bug fixes