Fixed .html() error, small fixes