adding Assert class usage and stylizing