almost working
fix style