reworked challenge 2, maybe it's not really working?