Small edits to support IE