Fixed errors

Testing code - fixed