seventh commit

refined code a small bit