Improvements

cc @dosyara