Enhancement/fixes