Move styling to appropriate element