Add simplecov && change code