Move setAttribute to element