added new lines of code

where there were none before