update - fix shitty coding style , don't get too clever