Added testing functionality to project
Fixed testing functionality part