made small improvements