chore(cleanup): Cleanup styles and remove unnecessary code