Finished writting tests