Improvements

Signed-off-by: Andrea Cappelli <a.cappelli87@gmail.com>