Update file to unit testing