Improvements to loki support

Signed-off-by: task650 <task650@gmail.com>