fix minor details in code