code not yet working properly