bitstarter enhancement from hw4