fixed few bugs, yolos.