quick attempt at code