little fixes in code