Added missing argument
Added comment
Fixed broken exceptions by importing class