Done on my own after watching solution