Small edits to main img code