fixed error and added style tweak