Style elements + add save callback