Working but have some troubles