add documentation and get rid of commented code
add additional dataset for testing