fixed few lines