Few fixes, also added melody note probability