quick update to fix minor bugs