[TASK] Add Tests to package