trying to fix code

Code needs more fixing Kaj