reworking code on to fix bug