updated test and fixed some lint issues