Finished with rough code