clean up css and add heart